import numpy as np
from pycocotools import mask

# 创建一个简单的二值掩码
binary_mask = np.array([
    [0, 0, 1, 1, 0],
    [0, 1, 1, 0, 0],
    [0, 0, 0, 0, 0]
], dtype=np.uint8)

# 将二值掩码转换为COCO RLE格式
rle = mask.encode(np.asfortranarray(binary_mask))

# 打印RLE编码
print("RLE编码：", rle)

# 从RLE解码回二值掩码
decoded_mask = mask.decode(rle)

# 打印解码后的掩码
print("解码后的掩码：")
print(decoded_mask)
